package com.spzx.product.service.impl;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.spzx.common.core.utils.StringUtils;
import com.spzx.product.api.domain.SkuStock;
import com.spzx.product.api.domain.dto.OrderStockDTO;
import com.spzx.product.mapper.SkuStockMapper;
import com.spzx.product.service.SkuStockService;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.List;
import java.util.stream.Collectors;

@Service
public class SkuStockServiceImpl extends ServiceImpl<SkuStockMapper, SkuStock> implements SkuStockService {

    @Transactional(rollbackFor = Exception.class)
    @Override
    public List<OrderStockDTO> checkAndLockStock(List<OrderStockDTO> orderStockDTOS) {
        if (StringUtils.isEmpty(orderStockDTOS)) {
            return null;
        }

        //1、检查并锁定库存
        // 可加分布式锁 对每个效验锁定库存skuId加锁 减小锁的范围

        checkLockStock(orderStockDTOS);
//2、检查：如果有库存不足的，应该将所有锁定的库存释放掉(用户提交订单业务失败)
        //获取锁库存成功的数据集合
        List<OrderStockDTO> lockedStockDTOs = orderStockDTOS.stream().filter(orderStockDTO -> orderStockDTO.getFlag())
                .collect(Collectors.toList());

        if (!StringUtils.isEmpty(lockedStockDTOs) && lockedStockDTOs.size() < orderStockDTOS.size()) {
            //如果有锁定库存失败的数据，需要释放锁定的库存
            releaseLockStock(lockedStockDTOs);
        }
        //3、返回校验并设置了锁定状态的 库存对象集合
        return orderStockDTOS;
    }












    ///释放锁的库存
    private void releaseLockStock(List<OrderStockDTO> lockedStockDTOs) {
        lockedStockDTOs.forEach(lockedStockDTO -> {
            baseMapper.releaseLockStock(lockedStockDTO);
        });

    }

    ///锁定库存
    private void checkLockStock(List<OrderStockDTO> orderStockDTOS) {
        orderStockDTOS.forEach(orderStockDTO -> {
            boolean flag = baseMapper.checkLockStock(orderStockDTO);
            orderStockDTO.setFlag(flag);
        });

    }
}
